/*
 * test/test.hpp
 * Copyright 2013 Google Inc. All Rights Reserved.
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 */

#pragma once
#ifndef H_54E531F7_9154_454B_BEB9_257408429470
#define H_54E531F7_9154_454B_BEB9_257408429470

#include <exception>
#include <vector>
#include <cstring>
#include <cstdio>
#include <sstream>
#include <string>

namespace test {

    struct AssertFailedError : std::exception {
        ~AssertFailedError() throw() { }

        AssertFailedError(const char *filename, int line, const char *errmsg) :
                basename(_basename(filename)), line(line), errmsg(errmsg) { }

        const char *what() const throw() {
            if (not _what.size()) {
                std::ostringstream ss;
                ss << "assertion failed (" << basename << ":" << line;
                ss << ") " << errmsg;
                _what = ss.str();
            }
            return _what.c_str();
        }

        const char *basename;
        int line;
        const char *errmsg;

        mutable std::string _what;

        static const char *_basename(const char *filename) {
            const char *basename = filename + strlen(filename);
            while (basename != filename && *basename != '/') {
                basename -= 1;
            }
            return basename + 1;
        }
    };

    enum TestStatus {
        SUCCESS = 0 << 0,
        FAILED = 1 << 0,

        ASSERT_FAIL = FAILED | 1 << 1,
        EXCEPTION_UNCAUGHT = FAILED | 2 << 1,
        SIGNAL_UNCAUGHT = FAILED | 3 << 1,
        SIGNAL_SEGFAULT = SIGNAL_UNCAUGHT | 1 << 3,
        SIGNAL_ABORT = SIGNAL_UNCAUGHT | 2 << 3,
        SIGNAL_DIVZERO = SIGNAL_UNCAUGHT | 2 << 3,

        STATUS_MASK = 0x1F
    };

    struct TestBase {
        const char *name;
        TestStatus expected_status;

        virtual ~TestBase() { }

        TestBase(const char *, TestStatus);

        virtual void do_test() = 0;

        TestStatus run() {
            try {
                do_test();
                return SUCCESS;
            } catch (const AssertFailedError &e) {
                printf("!! %s\n", e.what());
                return ASSERT_FAIL;
            } catch (const std::exception &e) {
                printf("!! exception: %s\n", e.what());
                return EXCEPTION_UNCAUGHT;
            } catch (...) {
                printf("!! unknown exception\n");
                return EXCEPTION_UNCAUGHT;
            }
        }
    };

    typedef std::vector<TestBase *> test_registry_t;
    extern test_registry_t test_registry;

    TestBase::TestBase(const char *n, TestStatus s) : name(n), expected_status(s) {
        test_registry.push_back(this);
    }

} // namespace test

#define _TEST_STATUS(name, status) \
    struct TEST_##name: ::test::TestBase { \
        TEST_##name(): TestBase(#name, status) {} \
        void do_test(); \
    } TEST_##name; \
    void TEST_##name::do_test()

#define TEST(name) _TEST_STATUS(name, ::test::SUCCESS)
#define TEST_FAIL(name) _TEST_STATUS(name, ::test::FAILED)
#define TEST_FAIL_ASSERT(name) _TEST_STATUS(name, ::test::ASSERT_FAIL)
#define TEST_UNCAUGHT_EXCEPTION(name) _TEST_STATUS(name, ::test::EXCEPTION_UNCAUGHT)
#define TEST_UNCAUGHT_SIGNAL(name) _TEST_STATUS(name, ::test::SIGNAL_UNCAUGHT)
#define TEST_SEGFAULT(name) _TEST_STATUS(name, ::test::SIGNAL_SEGFAULT)
#define TEST_ABORT(name) _TEST_STATUS(name, ::test::SIGNAL_ABORT)
#define TEST_DIVZERO(name) _TEST_STATUS(name, ::test::SIGNAL_DIVZERO)

#define ASSERT(expr) \
    (expr) ? static_cast<void>(0) \
    : throw ::test::AssertFailedError( \
            __FILE__, __LINE__, #expr)

#define _ASSERT_BINOP(a, b, cmp) \
    (not (a cmp b)) ? static_cast<void>(0) \
    : throw ::test::AssertFailedError( \
            __FILE__, __LINE__, "because " #a " " #cmp " " #b)

#define ASSERT_EQ(a, b) _ASSERT_BINOP(a, b, !=)
#define ASSERT_NE(a, b) _ASSERT_BINOP(a, b, ==)
#define ASSERT_LT(a, b) _ASSERT_BINOP(a, b, >=)
#define ASSERT_LE(a, b) _ASSERT_BINOP(a, b, >)
#define ASSERT_GT(a, b) _ASSERT_BINOP(a, b, <=)
#define ASSERT_GE(a, b) _ASSERT_BINOP(a, b, <)

#define ASSERT_THROW(expr, e_type) \
    do { try { expr } \
        catch (const e_type&) { break; } \
        throw ::test::AssertFailedError( \
                __FILE__, __LINE__, "expected exception " #e_type); \
    } while(0)

#define ASSERT_ANY_THROW(expr) \
    do { try { expr } \
        catch (...) { break; } \
        throw ::test::AssertFailedError( \
                __FILE__, __LINE__, "expected any exception"); \
    } while(0)

#define ASSERT_NO_THROW(expr) \
    try { expr } \
    catch (...) { \
        throw ::test::AssertFailedError( \
                __FILE__, __LINE__, "no exception expected"); \
    }

#endif /* H_GUARD */
